/*
 * Copyright (c) Huawei Technologies Co., Ltd. 2022-2023. All rights reserved.
 */
#ifndef ADD_CUSTOM_TILING_H
#define ADD_CUSTOM_TILING_H
#include "register/tilingdata_base.h"

namespace optiling {
BEGIN_TILING_DATA_DEF(TilingData)

  TILING_DATA_FIELD_DEF(int32_t, total_size);
  TILING_DATA_FIELD_DEF(int32_t, batch_size);
  TILING_DATA_FIELD_DEF(int32_t, batch_number);
  TILING_DATA_FIELD_DEF(int32_t, stride_size);
  TILING_DATA_FIELD_DEF(int32_t, stride_number);
  TILING_DATA_FIELD_DEF(int32_t, reduce_size);
  TILING_DATA_FIELD_DEF(int32_t, reduce_number);
  TILING_DATA_FIELD_DEF(int32_t, mode);



  // TILING_DATA_FIELD_DEF(int32_t, outputShapeDim);
END_TILING_DATA_DEF;

REGISTER_TILING_DATA_CLASS(Softmax, TilingData)
}
#endif // ADD_CUSTOM_TILING_H